import numpy  as np

input = np.array([[[[0,1,2],
              [3,4,5],
              [6,7,8]]]])

import pdb
pdb.set_trace()